%load_ext autoreload
%autoreload 2
from paths import *
%matplotlib inline
import pandas as pd
import seaborn as sns
import statsmodels.formula.api as smf
from sklearn.linear_model import LinearRegression
from sklearn import metrics
from sklearn.cross_validation import train_test_split
import numpy as np
from IPython.display import set_matplotlib_formats
set_matplotlib_formats('jpeg')
%config InlineBackend.figure_format = 'jpeg'
%config InlineBackend.rc = {'font.size': 10, 'figure.figsize': (6.0, 4.0), \
'figure.facecolor': (1, 1, 1, 0), 'savefig.dpi': 36, \
'figure.subplot.bottom': 0.125, 'figure.edgecolor': (1, 1, 1, 0)}
%%javascript
IPython.OutputArea.auto_scroll_threshold = 1000;
# instantiate utility class
gr = Graph()
mice = loadDataLight()
# gr.plotTF(mouseA, mask=False)
df = pd.DataFrame(columns=['mouse', 'exp', 'rep', 'cor', 'corR', 'corH', 'corR+corH'])
i=0
for mouse in mice.mice:
for exp in mouse.experiments:
for rep in exp.repeats:
rep.getModels(mouseA.experiments[2].repeats[0].alpha)
df.loc[i] = [mouse.name, int(exp.number), int(rep.number), rep.cor, rep.rmsR, rep.rmsH, rep.rms]
i+=1
df['mouse'] = df['mouse'].astype('category')
df['exp'] = df['exp'].astype('int')
df['rep'] = df['rep'].astype('int')
# plt.plot(mouseB.experiments[2].meanAlphaModel)
gr.fRmin = 0
gr.fHmin = 0
gr.fHmax = 1400
gr.fRmax = 100
gr.plotModels(mouseM1223M, 0,0, mouseB.experiments[2].meanAlphaModel)
gr.plotModels(mouseM1223M, 0,0, mouseM1223M.experiments[0].repeats[0].alpha,
title=r'$\alpha$ FROM SAME SIGNAL')
p_est = [0.4121683, 1.09384498, 3.68435018, 0.98029149]
x = np.real(xax(mouseM1217F.experiments[0].repeats[0].mRatio, 60000))
alpha = [guess_function(xi,p_est[0],p_est[1],p_est[2]) for xi in x]
gr.plotModels(mouseM1223M, 0,0, alpha)
df['mouse_cat'] = df.mouse.cat.codes
df[['mouse', 'mouse_cat']].drop_duplicates()
g = sns.FacetGrid(df, col="mouse_cat", size=4, aspect=.8)
g = g.map(sns.boxplot, "exp", "cor")
g.map(plt.axhline, y=0, ls=":", c="red")
plt.savefig(DIRECTORY + 'data-correlations.png')
alpha = mouseB.experiments[2].meanAlphaModel
plt.plot(xax(alpha, len(alpha)*20), alpha)
plt.title('Model of transfer function chosen')
plt.savefig(DIRECTORY + 'alpha.png', dpi=300)
g = sns.FacetGrid(df, col="mouse_cat", size=4, aspect=.8)
g = g.map(sns.boxplot, "exp", "corR+corH")
g.map(plt.axhline, y=0, ls=":", c="red")
g = sns.FacetGrid(df, col="mouse_cat", size=4, aspect=.8)
g = g.map(sns.boxplot, "exp", "corR")
g.map(plt.axhline, y=0, ls=":", c="red")
g = sns.FacetGrid(df, col="mouse_cat", size=4, aspect=.8)
g = g.map(sns.boxplot, "exp", "corH")
g.map(plt.axhline, y=0, ls=":", c="red")
plt.savefig(DIRECTORY + 'model-correlations.png')